package service

import (
    "errors"

    "github.com/go-admin-team/go-admin-core/sdk/service"
    "gorm.io/gorm"

    "go-admin/app/hunter/models"
    "go-admin/app/hunter/service/dto"
    "go-admin/common/actions"
    cDto "go-admin/common/dto"
)

type HtContract struct {
    service.Service
}

// GetPage 获取HtContract列表
func (e *HtContract) GetPage(c *dto.HtContractGetPageReq, p *actions.DataPermission, list *[]models.HtContract, count *int64) error {
    var err error
    var data models.HtContract

    err = e.Orm.Model(&data).
        Scopes(
            cDto.MakeCondition(c.GetNeedSearch()),
            cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
            actions.Permission(data.TableName(), p),
        ).Preload("Talent").
        Find(list).Limit(-1).Offset(-1).
        Count(count).Error
    if err != nil {
        e.Log.Errorf("HtContractService GetPage error:%s \r\n", err)
        return err
    }
    return nil
}

// Get 获取HtContract对象
func (e *HtContract) Get(d *dto.HtContractGetReq, p *actions.DataPermission, model *models.HtContract) error {
    var data models.HtContract

    err := e.Orm.Model(&data).
        Scopes(
            actions.Permission(data.TableName(), p),
        ).Preload("Talent").
        First(model, d.GetId()).Error
    if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
        err = errors.New("查看对象不存在或无权查看")
        e.Log.Errorf("Service GetHtContract error:%s \r\n", err)
        return err
    }
    if err != nil {
        e.Log.Errorf("db error:%s", err)
        return err
    }
    return nil
}

// Insert 创建HtContract对象
func (e *HtContract) Insert(c *dto.HtContractInsertReq) error {
    var err error
    var data models.HtContract
    c.Generate(&data)
    err = e.Orm.Create(&data).Error
    if err != nil {
        e.Log.Errorf("HtContractService Insert error:%s \r\n", err)
        return err
    }
    return nil
}

// Update 修改HtContract对象
func (e *HtContract) Update(c *dto.HtContractUpdateReq, p *actions.DataPermission) error {
    var err error
    var data = models.HtContract{}
    e.Orm.Scopes(
        actions.Permission(data.TableName(), p),
    ).First(&data, c.GetId())
    c.Generate(&data)

    db := e.Orm.Save(&data)
    if err = db.Error; err != nil {
        e.Log.Errorf("HtContractService Save error:%s \r\n", err)
        return err
    }
    if db.RowsAffected == 0 {
        return errors.New("无权更新该数据")
    }
    return nil
}

// Remove 删除HtContract
func (e *HtContract) Remove(d *dto.HtContractDeleteReq, p *actions.DataPermission) error {
    var data models.HtContract

    db := e.Orm.Model(&data).
        Scopes(
            actions.Permission(data.TableName(), p),
        ).Delete(&data, d.GetId())
    if err := db.Error; err != nil {
        e.Log.Errorf("Service RemoveHtContract error:%s \r\n", err)
        return err
    }
    if db.RowsAffected == 0 {
        return errors.New("无权删除该数据")
    }
    return nil
}
